#include "mmqtt.h"
#include "../../sys/sys.h"
#include "esp_log.h"

esp_mqtt_client_handle_t client;
/**
 * @brief  Event handler registered to receive MQTT events
 * 
 * @param message 
 * @param error_code 
 */
static void log_error_if_nonzero(const char *message, int error_code)
{
    if (error_code != 0) {
        ESP_LOGE(TAG, "Last error %s: 0x%x", message, error_code);
    }
}

static esp_mqtt5_user_property_item_t user_property_arr[] = {//键值对
        {"board", "esp32"},
        {"u", "xqyr"},
        {"p", "qwer1234"}
    };

#define USE_PROPERTY_ARR_SIZE   sizeof(user_property_arr)/sizeof(esp_mqtt5_user_property_item_t)

static esp_mqtt5_publish_property_config_t publish_property = {
    .payload_format_indicator = 1,//UTF8 编码
    .message_expiry_interval = 1000,//消息过期时间1000s
    .topic_alias = 0,//主题别名。用于减少传输的主题字符串长度。在这里设置为 0，表示不使用主题别名。
    .response_topic = "/topic/esp32/replay",//响应主题：当客户端收到此消息后，可以在这个主题上发布响应消息。
    .correlation_data = "replay",//，以便响应方可以识别请求。
    .correlation_data_len = 6,//：相关数据的长度。表示 correlation_data 的长度为5个字节。
};

static esp_mqtt5_subscribe_property_config_t subscribe_property = {
    .subscribe_id = 25555,//定阅标识符。用于标识特定的订阅请求。
    .no_local_flag = false,//。设置为 false 表示客户端在接收到自己发布的消息时也会触发相应的回调。如果设置为 true，则客户端不会收到自己发布的消息。
    .retain_as_published_flag = false,//设置为 false 表示当服务器发送保留消息时，它不会将消息标志（retain flag）传递给订阅者。如果设置为 true，则订阅者会收到带有保留标志的消息。
    .retain_handle = 0,//0：保留消息的处理方式。通常用于指示如何处理保留消息。
    .is_share_subscribe = true,//设置为 true 表示这是一个共享订阅。共享订阅允许多个客户端订阅同一个主题，并且负载会在这些客户端之间分发。
     .share_name = "group1",//在共享订阅中，所有订阅该主题的客户端都会加入这个共享组，服务器会将消息负载分发给组中的一个客户端。
};

static esp_mqtt5_subscribe_property_config_t subscribe1_property = {
    .subscribe_id = 25555,
    .no_local_flag = true,
    .retain_as_published_flag = false,
    .retain_handle = 0,
};

// static esp_mqtt5_unsubscribe_property_config_t unsubscribe_property = {
//     .is_share_subscribe = true,
//     .share_name = "group1",
// };

static esp_mqtt5_disconnect_property_config_t disconnect_property = {
    .session_expiry_interval = 60,
    .disconnect_reason = 0,
};
/**
 * @brief 打印用户属性
 * 
 * @param user_property 
 */
static void print_user_property(mqtt5_user_property_handle_t user_property)
{
    if (user_property) {
        uint8_t count = esp_mqtt5_client_get_user_property_count(user_property);
        if (count) {
            esp_mqtt5_user_property_item_t *item = malloc(count * sizeof(esp_mqtt5_user_property_item_t));
            if (esp_mqtt5_client_get_user_property(user_property, item, &count) == ESP_OK) {
                for (int i = 0; i < count; i ++) {
                    esp_mqtt5_user_property_item_t *t = &item[i];
                    ESP_LOGI(TAG, "key is %s, value is %s", t->key, t->value);
                    free((char *)t->key);
                    free((char *)t->value);
                }
            }
            free(item);
        }
    }
}

/**
 * @brief MQTT 事件 处理
 * 
 * @param handler_args 
 * @param base 
 * @param event_id 
 * @param event_data 
 */
static void mqtt5_event_handler(void *handler_args, esp_event_base_t base, int32_t event_id, void *event_data)
{
    ESP_LOGD(TAG, "Event dispatched from event loop base=%s, event_id=%" PRIi32, base, event_id);
    esp_mqtt_event_handle_t event = event_data;
    esp_mqtt_client_handle_t client = event->client;
    int msg_id;

    ESP_LOGD(TAG, "free heap size is %" PRIu32 ", minimum %" PRIu32, esp_get_free_heap_size(), esp_get_minimum_free_heap_size());
    switch ((esp_mqtt_event_id_t)event_id) {
    case MQTT_EVENT_CONNECTED://链接成功
        ESP_LOGI(TAG, "MQTT_EVENT_CONNECTED");
        print_user_property(event->property->user_property);//打印连接成功事件中包含的用户属性。
        //函数设置发布消息的用户属性，
        esp_mqtt5_client_set_user_property(&publish_property.user_property, user_property_arr, USE_PROPERTY_ARR_SIZE);
        //设置为MQTT客户端的发布属性。
        esp_mqtt5_client_set_publish_property(client, &publish_property);
        //参数 client 是MQTT客户端句柄。
        // 参数 "/topic/qos1" 是发布的主题。
        // 参数 "data_3" 是发布的消息内容。
        // 参数 0 表示消息长度（0表示自动计算长度）。
        // 参数 1 表示QoS等级为1。
        // 参数 1 表示消息需要保留（retain）。
        msg_id = esp_mqtt_client_publish(client, "/topic/temp", "data_3", 0, 0, 1);
        esp_mqtt5_client_delete_user_property(publish_property.user_property);
        publish_property.user_property = NULL;
        ESP_LOGI(TAG, "sent publish successful, msg_id=%d", msg_id);
//设置订阅消息的用户属性，将 user_property_arr 数组赋值给 subscribe_property 的 user_property 字段。
        esp_mqtt5_client_set_user_property(&subscribe_property.user_property, user_property_arr, USE_PROPERTY_ARR_SIZE);
        esp_mqtt5_client_set_subscribe_property(client, &subscribe_property);
        //这是一个用于向MQTT服务器发送订阅请求的函数。Q0s:0
        msg_id = esp_mqtt_client_subscribe(client, "/topic/qos0", 0);
        esp_mqtt5_client_delete_user_property(subscribe_property.user_property);
        subscribe_property.user_property = NULL;
        ESP_LOGI(TAG, "sent subscribe successful, msg_id=%d", msg_id);
//：使用 esp_mqtt5_client_set_user_property 函数将 user_property_arr 数组中的用户属性赋值给 subscribe1_property 的 user_property 字段。
//这样，在订阅请求中可以携带这些用户属性
        esp_mqtt5_client_set_user_property(&subscribe1_property.user_property, user_property_arr, USE_PROPERTY_ARR_SIZE);
//使用 esp_mqtt5_client_set_subscribe_property 
//函数将配置好的 subscribe1_property 设置为 MQTT 客户端的订阅属性。这样，订阅请求会携带这些配置属性。
        esp_mqtt5_client_set_subscribe_property(client, &subscribe1_property);
        msg_id = esp_mqtt_client_subscribe(client, "/topic/qos1", 2);
        esp_mqtt5_client_delete_user_property(subscribe1_property.user_property);
        subscribe1_property.user_property = NULL;
        ESP_LOGI(TAG, "sent subscribe successful, msg_id=%d", msg_id);

        // esp_mqtt5_client_set_user_property(&unsubscribe_property.user_property, user_property_arr, USE_PROPERTY_ARR_SIZE);
        // esp_mqtt5_client_set_unsubscribe_property(client, &unsubscribe_property);
        // msg_id = esp_mqtt_client_unsubscribe(client, "/topic/qos0");
        // ESP_LOGI(TAG, "sent unsubscribe successful, msg_id=%d", msg_id);
        // esp_mqtt5_client_delete_user_property(unsubscribe_property.user_property);
        // unsubscribe_property.user_property = NULL;
        break;
    case MQTT_EVENT_DISCONNECTED:
        ESP_LOGI(TAG, "MQTT_EVENT_DISCONNECTED");
        print_user_property(event->property->user_property);
        break;
    case MQTT_EVENT_SUBSCRIBED:
        ESP_LOGI(TAG, "MQTT_EVENT_SUBSCRIBED, msg_id=%d", event->msg_id);
        print_user_property(event->property->user_property);
        esp_mqtt5_client_set_publish_property(client, &publish_property);
        msg_id = esp_mqtt_client_publish(client, "/topic/qos0", "qoso", 0, 0, 0);
        ESP_LOGI(TAG, "sent publish successful, msg_id=%d", msg_id);
        break;
    case MQTT_EVENT_UNSUBSCRIBED:
        ESP_LOGI(TAG, "MQTT_EVENT_UNSUBSCRIBED, msg_id=%d", event->msg_id);
        print_user_property(event->property->user_property);
        esp_mqtt5_client_set_user_property(&disconnect_property.user_property, user_property_arr, USE_PROPERTY_ARR_SIZE);
        esp_mqtt5_client_set_disconnect_property(client, &disconnect_property);
        esp_mqtt5_client_delete_user_property(disconnect_property.user_property);
        disconnect_property.user_property = NULL;
        esp_mqtt_client_disconnect(client);
        break;
    case MQTT_EVENT_PUBLISHED:
        ESP_LOGI(TAG, "MQTT_EVENT_PUBLISHED, msg_id=%d", event->msg_id);
        print_user_property(event->property->user_property);
        break;
    case MQTT_EVENT_DATA:
        ESP_LOGI(TAG, "MQTT_EVENT_DATA");
        print_user_property(event->property->user_property);
        // ESP_LOGI(TAG, "payload_format_indicator is %d", event->property->payload_format_indicator);
        // ESP_LOGI(TAG, "response_topic is %.*s", event->property->response_topic_len, event->property->response_topic);
        // ESP_LOGI(TAG, "correlation_data is %.*s", event->property->correlation_data_len, event->property->correlation_data);
        // ESP_LOGI(TAG, "content_type is %.*s", event->property->content_type_len, event->property->content_type);
        ESP_LOGI(TAG, "TOPIC=%.*s", event->topic_len, event->topic);
        ESP_LOGI(TAG, "DATA=%.*s", event->data_len, event->data);
        break;
    case MQTT_EVENT_ERROR:
        ESP_LOGI(TAG, "MQTT_EVENT_ERROR");
        print_user_property(event->property->user_property);
        ESP_LOGI(TAG, "MQTT5 return code is %d", event->error_handle->connect_return_code);
        if (event->error_handle->error_type == MQTT_ERROR_TYPE_TCP_TRANSPORT) {
            log_error_if_nonzero("reported from esp-tls", event->error_handle->esp_tls_last_esp_err);
            log_error_if_nonzero("reported from tls stack", event->error_handle->esp_tls_stack_err);
            log_error_if_nonzero("captured as transport's socket errno",  event->error_handle->esp_transport_sock_errno);
            ESP_LOGI(TAG, "Last errno string (%s)", strerror(event->error_handle->esp_transport_sock_errno));
        }
        break;
    default:
        ESP_LOGI(TAG, "Other event id:%d", event->event_id);
        break;
    }
}

void mqtt5_app_start(void)
{

    esp_mqtt_client_config_t mqtt5_cfg = {
        .broker.address.uri = CONFIG_BROKER_URL,
        .broker.address.port = 1883,//端口
        .session.protocol_ver = MQTT_PROTOCOL_V_5,//版本
        .network.disable_auto_reconnect = true,//启用自动重连
        .credentials.username = "xqyr",//用户名
        .credentials.authentication.password = "qwer1234",//密码，没有
        .session.last_will.topic = "/topic/esp32_dh11_topic",//异常断开发送消息(遗愿消息)
        .session.last_will.msg = "disconnct",//内容
        .session.last_will.msg_len = sizeof("disconnct"),
        .session.last_will.qos = 0,//
        .session.last_will.retain = true,//遗愿消息是否保留
    };
    //用于配置 MQTT 连接的属性
    esp_mqtt5_connection_property_config_t connect_property = {
        .session_expiry_interval = 10,//会话过期时间，单位为秒
        .maximum_packet_size = 1024,//最大数据包
        .receive_maximum = 65535,//接收消息最大数量
        .topic_alias_maximum = 2,//主题别名最大值，用于减少传输的主题字符串长度。在这里设置为 2，表示客户端最多可以使用两个主题别名。
        .request_resp_info = true,//客户端请求包含响应信息
        .request_problem_info = true,//客户端请求包含问题信息
        .will_delay_interval = 10,//遗愿消息延迟间隔，单位为秒。在这里设置为 10 秒，表示遗愿消息发送的延迟时间。
        .payload_format_indicator = true,//负载格式指示器。设置为 true，表示负载的格式由客户端指示。
        .message_expiry_interval = 10,//消息过期间隔，单位为秒。在这里设置为 10 秒，表示消息的有效期限。
        .response_topic = "/test/response_topic",//响应主题。在这里设置为 "/test/response"，表示用于响应的主题名称。
        .correlation_data = "correlation_data",//相关数据，用于关联不同消息之间的关系。在这里设置为 "123456"，是一个字符串形式的相关数据。
        .correlation_data_len = sizeof("correlation_data"),//相关数据长度。在这里设置为 6，表示相关数据字符串的长度为6个字节。
    };

#if CONFIG_BROKER_URL_FROM_STDIN
    char line[128];

    if (strcmp(mqtt5_cfg.uri, "FROM_STDIN") == 0) {
        int count = 0;
        printf("Please enter url of mqtt broker\n");
        while (count < 128) {
            int c = fgetc(stdin);
            if (c == '\n') {
                line[count] = '\0';
                break;
            } else if (c > 0 && c < 127) {
                line[count] = c;
                ++count;
            }
            vTaskDelay(10 / portTICK_PERIOD_MS);
        }
        mqtt5_cfg.broker.address.uri = line;
        printf("Broker url: %s\n", line);
    } else {
        ESP_LOGE(TAG, "Configuration mismatch: wrong broker url");
        abort();
    }
#endif /* CONFIG_BROKER_URL_FROM_STDIN */
//初始化mqtt客户端
    client = esp_mqtt_client_init(&mqtt5_cfg);

    /* Set connection properties and user properties */
    esp_mqtt5_client_set_user_property(&connect_property.user_property, user_property_arr, USE_PROPERTY_ARR_SIZE);
    esp_mqtt5_client_set_user_property(&connect_property.will_user_property, user_property_arr, USE_PROPERTY_ARR_SIZE);
    esp_mqtt5_client_set_connect_property(client, &connect_property);

    /* If you call esp_mqtt5_client_set_user_property to set user properties, DO NOT forget to delete them.
     * esp_mqtt5_client_set_connect_property will malloc buffer to store the user_property and you can delete it after
     */
    esp_mqtt5_client_delete_user_property(connect_property.user_property);
    esp_mqtt5_client_delete_user_property(connect_property.will_user_property);

    /* The last argument may be used to pass data to the event handler, in this example mqtt_event_handler */
    esp_mqtt_client_register_event(client, ESP_EVENT_ANY_ID, mqtt5_event_handler, NULL);//注册了一个事件处理函数 mqtt5_event_handler。该函数将处理所有MQTT事件（ESP_EVENT_ANY_ID 表示所有事件）。
    esp_mqtt_client_start(client);
}
